# Set Parameters for Algorithms

# Lasso
lasso_params = list(alpha = 0.95,
                    family = "binomial")

lasso_count_params = list(alpha = 0.95,
                          family = "poisson")

lasso_demean_params = list(alpha = 0.95,
                           family = "gaussian")

# GBM
gbm_params = list(distribution="bernoulli",
                  interaction.depth = 3,
                  n.minobsinnode = 10,
                  verbose = FALSE,
                  shrinkage = .1,
                  train.fraction = 1,
                  bag.fraction = 1
)

gbm_count_params = list(distribution="poisson",
                        interaction.depth = 3,
                        n.minobsinnode = 10,
                        verbose = FALSE,
                        shrinkage = .1,
                        train.fraction = 1,
                        bag.fraction = 1
)

gbm_demean_params = list(distribution="gaussian",
                         interaction.depth = 3,
                         n.minobsinnode = 10,
                         verbose = FALSE,
                         shrinkage = .1,
                         train.fraction = 1,
                         bag.fraction = 1
)

# Random Forests
rf_params = list(ntree = 100,
                 maxnodes = 60,
                 type = 'regression',
                 importance = FALSE)
rf_count_params = list(ntree = 500,
                       nodesize = 30, 
                       type = 'regression',
                       importance = FALSE,
                       do.trace = FALSE
)
rf_demean_params = list(ntree = 100,
                       nodesize = 15, 
                       mtry = 20,
                       type = 'regression',
                       importance = FALSE,
                       do.trace = FALSE
)

# Neural Networks
nn_params = list(pcNum = list(indo=30,
                              colo=20,
                              colo_ext1=20,
                              colo_ext2=20),
                 decay = .15,
                 trace = FALSE)

nn_count_params = list(pcNum = 20,
                       decay = .15,
                       trace = FALSE,
                       linout = TRUE
)
nn_demean_params = nn_count_params